package com.it.od.lee;

import java.util.ArrayList;
import java.util.List;

/**
 * @author: liminghui
 * @date: 2024/3/29 21:12
 * @version: 1.0
 * @description: 组合
 * 给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。
 * 示例 1:
 * 输入: n = 4, k = 2
 * 输出:
 * [
 * [2,4],
 * [3,4],
 * [2,3],
 * [1,2],
 * [1,3],
 * [1,4],
 * ]
 * 示例 2:
 * <p>
 * 输入: n = 1, k = 1
 * 输出: [[1]]
 */
public class LCR80 {
    public static void main(String[] args) {
        int n = 4;
        int k = 2;
        List<List<Integer>> res = combine(n, k);
        System.out.println(res);
    }

    public static List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> ans = new ArrayList<>();
        List<Integer> candidate = new ArrayList<>();
        dfs(n, k, 1, candidate, ans);
        return ans;
    }

    private static void dfs(int n, int k, int idx, List<Integer> candidate, List<List<Integer>> ans) {
        if (candidate.size() == k) { // 满足结果
            ans.add(new ArrayList<>(candidate));
            return;
        }
        // 优化剪枝 如果已经选用的个数+剩余数个数小于 k，则不能选了、。
        if (candidate.size() + (n - idx + 1) < k) {
            return;
        }

        for (int i = idx; i <= n; i++) {
            candidate.add(i);
            dfs(n, k, i + 1, candidate, ans);
            candidate.remove(candidate.size() - 1);
        }
    }
}
